package com.example.leetcode;

/**
 * 给定两个大小分别为 m 和 n 的正序（从小到大）数组nums1 和nums2。请你找出并返回这两个正序数组的 中位数 。
 * 算法的时间复杂度应该为 O(log (m+n)) 。
 *
 * @link { <a href="https://leetcode.cn/problems/median-of-two-sorted-arrays">...</a> }
 */
public class Code04 {
    public static void main(String[] args) {
        Code04 code04 = new Code04();
        int[] arr1 = new int[]{1, 2};
        int[] arr2 = new int[]{3, 4};
        System.out.println(code04.findMedianSortedArrays(arr1, arr2));

    }

    // 数组合并
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int size = nums1.length + nums2.length;
        int[] arr = new int[size];
        int i = 0, j = 0, k = 0;
        while (i < nums1.length && j < nums2.length) {
            if (nums1[i] < nums2[j]) {
                arr[k++] = nums1[i++];
            } else {
                arr[k++] = nums2[j++];
            }
        }
        while (i < nums1.length) {
            arr[k++] = nums1[i++];
        }
        while (j < nums2.length) {
            arr[k++] = nums2[j++];
        }
        // 注意除法
        return size % 2 == 0 ? ((arr[size / 2] + arr[(size / 2 - 1)]) / 2.0) : arr[size / 2];
    }

    // 数组合并
    public double findMedianSortedArrays1(int[] nums1, int[] nums2) {
        int size = nums1.length + nums2.length;
        int[] arr = new int[size];
        int i = 0, j = 0, k = 0;
        while (i < nums1.length && j < nums2.length) {
            if (nums1[i] < nums2[j]) {
                i++;
            } else {
                j++;
            }
            k++;
        }
        while (i < nums1.length) {
            k++;
            i++;
        }
        while (j < nums2.length) {
            k++;
            j++;
        }
        // 注意除法
        return size % 2 == 0 ? ((arr[size / 2] + arr[(size / 2 - 1)]) / 2.0) : arr[size / 2];
    }
}
